%module MNISTDataLoaderModule
%import "./LibDL/data/MNISTDataIterator/MNISTDataIterator.i"

%typemap(javainterfaces) data::MNISTDataLoader "java.lang.Iterable<LibDL.core.data.Example>"
%typemap(javaimports) SWIGTYPE
%{
import LibDL.core.*;
%}

%{
#include "./LibDL/data/MNISTDataLoader/MNISTDataLoader.h"
%}

%nspace data::MNISTDataLoader;
%include "./LibDL/data/MNISTDataLoader/MNISTDataLoader.h"